Machine learning model abstraction layer for runtime efficiency

ABSTRACT

Systems and methods include receiving a trained machine learning model that has been processed with training information removed therefrom, wherein the training information is utilized in training of the trained machine learning model; monitoring traffic, inline at the node, including processing the traffic with the trained machine learning model; obtaining a verdict on the traffic based on the trained machine learning model; and performing an action on the traffic based on the verdict.

CROSS-REFERENCE TO RELATED APPLICATION(S)

The present disclosure is a continuation of U.S. patent application Ser.No. 17/024,762, filed Sep. 18, 2020, the contents of which areincorporated by reference in their entirety. U.S. patent applicationSer. No. 17/024,762 is a continuation-in-part of the following U.S.patent applications, the contents of which are incorporated by referencein their entirety:

application Ser. No. Filing Date Title 16/377,129 Apr. 5, 2019 Prudentensemble models in machine learning with high precision for use innetwork security 16/902,759 Jun. 16, 2020 Utilizing Machine Learning forsmart quarantining of potentially malicious files

FIELD OF THE DISCLOSURE

The present disclosure relates generally to networking and computing.More particularly, the present disclosure relates to systems and methodsfor a Machine Learning (ML) model abstraction layer for runtimeefficiency.

BACKGROUND OF THE DISCLOSURE

Machine Learning (ML) techniques are proliferating and offer many usecases. In network and computer security, use cases for machine learninginclude malware detection, identifying malicious files for furtherprocessing such as in a sandbox, user or content risk determination,intrusion detection, anomalous behavior detection, etc. That is, thereare various possible use cases for machine learning in network andcomputer security. One important aspect of using a machine learningmodel in production (runtime) in network and computer security is a needfor speed, processing efficiency, and minimal latency. This aspect iscritical, especially in cloud-based services. The general process inmachine learning includes training where a machine learning model istrained on a dataset, e.g., data including malicious and benign contentor files, and, once trained, the machine learning model is used inproduction (runtime) to classify unknown content based on the training.That is, the machine learning process includes a training step and aproduction step. In production, the machine learning model should belean and purpose-built for runtime.

BRIEF SUMMARY OF THE DISCLOSURE

The present disclosure relates to systems and methods for a MachineLearning (ML) model abstraction layer for runtime efficiency.Specifically, the present disclosure provides a model abstraction layerthat is used to serve a machine learning model in production. The modelabstraction layer includes removing training information that is notrelevant to runtime, e.g., hyperparameters. The model abstraction layeris algorithm and programming language agnostic, supporting anyarchitecture. With this approach, the training process is decoupled fromthe runtime process, leading to a lean, purpose-built model for runtime.

The present disclosure includes a method including steps, an apparatusconfigured to implement steps, and a non-transitory computer-readablestorage medium having computer-readable code stored thereon forprogramming one or more processors to perform the steps. The stepsinclude training a machine learning model with data for identifyingfeatures in monitored traffic in a network; analyzing the trainedmachine learning model to identify information overhead therein, whereinthe information overhead is utilized in part for the training; removingthe information overhead in the machine learning model; and providingthe machine learning model for runtime use for identifying the featuresin the monitored traffic, with the removed information overhead from themachine learning model.

The steps can further include identifying features that are not used inthe trained machine learning model; and removing the identified featuresprior to the providing. The steps can further include determiningthresholds for the identifying features in the trained machine learningmodel; and normalizing the thresholds to a scoring system. Theinformation overhead can include hyperparameters. The informationoverhead can include metadata that is extra overhead in the trainedmachine learning model. The information overhead can include informationfrom the training that is not useful at runtime in the trained machinelearning model. The information overhead can include parameters that areused to understand the training. The machine learning model can includea gradient boosting framework that uses tree-based learning algorithms.The providing can be to a cloud-based system that utilizes the machinelearning model for inline monitoring of the monitored traffic.

In another embodiment, the steps include receiving a trained machinelearning model that has been processed with training information removedtherefrom, wherein the training information is utilized in training ofthe trained machine learning model; monitoring traffic, inline at thenode, including processing the traffic with the trained machine learningmodel; obtaining a verdict on the traffic based on the trained machinelearning model; and performing an action on the traffic based on theverdict. The verdict can be associated with malware detection and theaction includes whether or not to sandbox content in the traffic. Theverdict can be associated with intrusion detection and the actionincludes whether to allow or block some of the traffic. The verdict canbe associated with threat classification and the action includes whetherto allow or block content in the traffic.

The training information is not relevant to runtime, including featuresnot used in the trained machine learning model. The training informationcan include parameters that are useful at understanding how the trainedmachine learning model was trained, but serve no purpose at runtime. Thetraining information can include hyperparameters used to controltraining. The training information can include metadata that is extraoverhead in the trained machine learning model. The training informationcan include any of feature names, feature info, and tree sizes in thetrained machine learning model. The trained machine learning model caninclude a gradient boosting framework that uses tree-based learningalgorithms.

BRIEF DESCRIPTION OF THE DRAWINGS

The present disclosure is illustrated and described herein withreference to the various drawings, in which like reference numbers areused to denote like system components/method steps, as appropriate, andin which:

FIG. 1A is a network diagram of a cloud-based system offering securityas a service;

FIG. 1B is a network diagram of an example implementation of thecloud-based system;

FIG. 2A is a block diagram of a server that may be used in thecloud-based system of FIGS. 1A and 1B or the like;

FIG. 2B is a block diagram of a user device that may be used with thecloud-based system of FIGS. 1A and 1B or the like;

FIG. 3 is a diagram of a trained machine learning model in the form of adecision tree;

FIG. 4 is a flowchart of a prudent ensemble model process;

FIG. 5 is a flowchart of a content processing process, implemented bythe cloud-based security system;

FIG. 6 is a flow diagram of a conventional quarantine process forquarantining, scanning, blocking, and allowing a file;

FIG. 7 is a flow diagram of a smart quarantine process for quarantining,scanning, blocking, and allowing a file, where machine learning is usedto front end whether or not to hold the file;

FIG. 8 is a flow diagram of a smart quarantine process for quarantining,scanning, blocking, and allowing a file, where machine learning is usedto front end whether or not to hold or scan the file;

FIG. 9 is a flow diagram of a smart quarantine process for quarantining,scanning, blocking, and allowing a file, where machine learning is usedto front end whether or not to hold, scan, or allow the file;

FIG. 10 is a screenshot of an example of the metadata that is extraoverhead in a machine learning model;

FIG. 11 is a screenshot of an example of information that is not usefulat runtime in the machine learning model;

FIG. 12 is a screenshot of parameters that are useful at understandinghow the machine learning model was trained, but serve no purpose atruntime; and

FIG. 13 is a flowchart of a machine learning abstraction process.

DETAILED DESCRIPTION OF THE DISCLOSURE

Again, the present disclosure relates to systems and methods for aMachine Learning (ML) model abstraction layer for runtime efficiency.Specifically, the present disclosure provides a model abstraction layerthat is used to serve a machine learning model in production. The modelabstraction layer includes removing training information that is notrelevant to runtime, e.g., hyperparameters. The model abstraction layeris algorithm and programming language agnostic, supporting anyarchitecture. With this approach, the training process is decoupled fromthe runtime process, leading to a lean, purpose-built model for runtime.

Also, the present disclosure relates to systems and methods utilizingMachine Learning (ML) for smart quarantining of files, such as forscanning, sandboxing, etc. in a cloud-based system. Specifically, thepresent disclosure presents a smart quarantine with a goal of minimizingthe number of files quarantined, the number of malicious files passedthrough to an end user, and a number of files scanned by a sandbox. Inminimizing each of these metrics, the smart quarantine provides betterUX relative to conventional approaches, lower risk as the risky filesare only scanned/quarantined, and lower cost as the sandbox only focusedon files of interest as detected by machine learning. The presentdisclosure can be implemented in an antivirus program resident on a userdevice, in a standalone sandbox, in a security appliance, and/or througha cloud-based system offering security-as-a-service.

Example Cloud-Based System

FIG. 1A is a network diagram of a cloud-based system 100 offeringsecurity as a service. Specifically, the cloud-based system 100 canoffer a Secure Internet and Web Gateway as a service to various users102, as well as other cloud services. In this manner, the cloud-basedsystem 100 is located between the users 102 and the Internet as well asany cloud services 106 (or applications) accessed by the users 102. Assuch, the cloud-based system 100 provides inline monitoring inspectingtraffic between the users 102, the Internet 104, and the cloud services106, including Secure Sockets Layer (SSL) traffic. The cloud-basedsystem 100 can offer access control, threat prevention, data protection,etc. The access control can include a cloud-based firewall, cloud-basedintrusion detection, Uniform Resource Locator (URL) filtering, bandwidthcontrol, Domain Name System (DNS) filtering, etc. The threat preventioncan include cloud-based intrusion prevention, protection againstadvanced threats (malware, spam, Cross-Site Scripting (XSS), phishing,etc.), cloud-based sandbox, antivirus, DNS security, etc. The dataprotection can include Data Loss Prevention (DLP), cloud applicationsecurity such as via Cloud Access Security Broker (CASB), file typecontrol, etc.

The cloud-based firewall can provide Deep Packet Inspection (DPI) andaccess controls across various ports and protocols as well as beingapplication and user aware. The URL filtering can block, allow, or limitwebsite access based on policy for a user, group of users, or entireorganization, including specific destinations or categories of URLs(e.g., gambling, social media, etc.). The bandwidth control can enforcebandwidth policies and prioritize critical applications such as relativeto recreational traffic. DNS filtering can control and block DNSrequests against known and malicious destinations.

The cloud-based intrusion prevention and advanced threat protection candeliver full threat protection against malicious content such as browserexploits, scripts, identified botnets and malware callbacks, etc. Thecloud-based sandbox can block zero-day exploits (just identified) byanalyzing unknown files for malicious behavior. Advantageously, thecloud-based system 100 is multi-tenant and can service a large volume ofthe users 102. As such, newly discovered threats can be promulgatedthroughout the cloud-based system 100 for all tenants practicallyinstantaneously. The antivirus protection can include antivirus,antispyware, antimalware, etc. protection for the users 102, usingsignatures sourced and constantly updated. The DNS security can identifyand route command-and-control connections to threat detection enginesfor full content inspection.

The DLP can use standard and/or custom dictionaries to continuouslymonitor the users 102, including compressed and/or SSL-encryptedtraffic. Again, being in a cloud implementation, the cloud-based system100 can scale this monitoring with near-zero latency on the users 102.The cloud application security can include CASB functionality todiscover and control user access to known and unknown cloud services106. The file type controls enable true file type control by the user,location, destination, etc. to determine which files are allowed or not.

For illustration purposes, the users 102 of the cloud-based system 100can include a mobile device 110, a headquarters (H.Q.) 112 which caninclude or connect to a data center (DC) 114, Internet of Things (IoT)devices 116, a branch office/remote location 118, etc., and eachincludes one or more user devices (an example user device 250 isillustrated in FIG. 2B). The devices 110, 116, and the locations 112,114, 118 are shown for illustrative purposes, and those skilled in theart will recognize there are various access scenarios and other users102 for the cloud-based system 100, all of which are contemplatedherein. The users 102 can be associated with a tenant, which may includean enterprise, a corporation, an organization, etc. That is, a tenant isa group of users who share a common access with specific privileges tothe cloud-based system 100, a cloud service, etc. In an embodiment, theheadquarters 112 can include an enterprise's network with resources inthe data center 114. The mobile device 110 can be a so-called roadwarrior, i.e., users that are off-site, on-the-road, etc. Further, thecloud-based system 100 can be multi-tenant, with each tenant having itsown users 102 and configuration, policy, rules, etc. One advantage ofthe multi-tenancy and a large volume of users is the zero-day/zero-hourprotection in that a new vulnerability can be detected and theninstantly remediated across the entire cloud-based system 100. The sameapplies to policy, rule, configuration, etc. changes—they are instantlyremediated across the entire cloud-based system 100. As well, newfeatures in the cloud-based system 100 can also be rolled upsimultaneously across the user base, as opposed to selective andtime-consuming upgrades on every device at the locations 112, 114, 118,and the devices 110, 116.

Logically, the cloud-based system 100 can be viewed as an overlaynetwork between users (at the locations 112, 114, 118, and the devices110, 106) and the Internet 104 and the cloud services 106. Previously,the I.T. deployment model included enterprise resources and applicationsstored within the data center 114 (i.e., physical devices) behind afirewall (perimeter), accessible by employees, partners, contractors,etc. on-site or remote via Virtual Private Networks (VPNs), etc. Thecloud-based system 100 is replacing the conventional deployment model.The cloud-based system 100 can be used to implement these services inthe cloud without requiring the physical devices and management thereofby enterprise I.T. administrators. As an ever-present overlay network,the cloud-based system 100 can provide the same functions as thephysical devices and/or appliances regardless of geography or locationof the users 102, as well as independent of platform, operating system,network access technique, network access provider, etc.

There are various techniques to forward traffic between the users 102 atthe locations 112, 114, 118, and via the devices 110, 116, and thecloud-based system 100. Typically, the locations 112, 114, 118 can usetunneling where all traffic is forward through the cloud-based system100. For example, various tunneling protocols are contemplated, such asGeneric Routing Encapsulation (GRE), Layer Two Tunneling Protocol(L2TP), Internet Protocol (I.P.) Security (IPsec), customized tunnelingprotocols, etc. The devices 110, 116 can use a local application thatforwards traffic, a proxy such as via a Proxy Auto-Config (PAC) file,and the like. A key aspect of the cloud-based system 100 is all trafficbetween the users 102 and the Internet 104 or the cloud services 106 isvia the cloud-based system 100. As such, the cloud-based system 100 hasvisibility to enable various functions, all of which are performed offthe user device in the cloud.

The cloud-based system 100 can also include a management system 120 fortenant access to provide global policy and configuration as well asreal-time analytics. This enables I.T. administrators to have a unifiedview of user activity, threat intelligence, application usage, etc. Forexample, I.T. administrators can drill-down to a per-user level tounderstand events and correlate threats, to identify compromiseddevices, to have application visibility, and the like. The cloud-basedsystem 100 can further include connectivity to an Identity Provider(IDP) 122 for authentication of the users 102 and to a SecurityInformation and Event Management (SIEM) system 124 for event logging.The system 124 can provide alert and activity logs on a per-user 102basis.

FIG. 1B is a network diagram of an example implementation of thecloud-based system 100. In an embodiment, the cloud-based system 100includes a plurality of enforcement nodes (EN) 150, labeled asenforcement nodes 150-1, 150-2, 150-N, interconnected to one another andinterconnected to a central authority (CA) 152. The nodes 150, 152,while described as nodes, can include one or more servers, includingphysical servers, virtual machines (V.M.) executed on physical hardware,etc. That is, a single node 150, 152 can be a cluster of devices. Anexample of a server is illustrated in FIG. 2 . The cloud-based system100 further includes a log router 154 that connects to a storage cluster156 for supporting log maintenance from the enforcement nodes 150. Thecentral authority 152 provide centralized policy, real-time threatupdates, etc. and coordinates the distribution of this data between theenforcement nodes 150. The enforcement nodes 150 provide an onramp tothe users 102 and are configured to execute policy, based on the centralauthority 152, for each user 102. The enforcement nodes 150 can begeographically distributed, and the policy for each user 102 followsthat user 102 as he or she connects to the nearest (or other criteria)enforcement node 150.

The enforcement nodes 150 are full-featured secure internet gatewaysthat provide integrated internet security. They inspect all web trafficbi-directionally for malware and enforce security, compliance, andfirewall policies, as described herein. In an embodiment, eachenforcement node 150 has two main modules for inspecting traffic andapplying policies: a web module and a firewall module. The enforcementnodes 150 are deployed around the world and can handle hundreds ofthousands of concurrent users with millions of concurrent sessions.Because of this, regardless of where the users 102 are, they can accessthe Internet 104 from any device, and the enforcement nodes 150 protectthe traffic and apply corporate policies. The enforcement nodes 150 canimplement various inspection engines therein, and optionally, sendsandboxing to another system. The enforcement nodes 150 includesignificant fault tolerance capabilities, such as deployment inactive-active mode to ensure availability and redundancy as well ascontinuous monitoring.

In an embodiment, customer traffic is not passed to any other componentwithin the cloud-based system 100, and the enforcement nodes 150 can beconfigured never to store any data to disk. Packet data is held inmemory for inspection and then, based on policy, is either forwarded ordropped. Log data generated for every transaction is compressed,tokenized, and exported over secure TLS connections to the log routers154 that direct the logs to the storage cluster 156, hosted in theappropriate geographical region, for each organization.

The central authority 152 hosts all customer (tenant) policy andconfiguration settings. It monitors the cloud and provides a centrallocation for software and database updates and threat intelligence.Given the multi-tenant architecture, the central authority 152 isredundant and backed up in multiple different data centers. Theenforcement nodes 150 establish persistent connections to the centralauthority 152 to download all policy configurations. When a new userconnects to an enforcement node 150, a policy request is sent to thecentral authority 152 through this connection. The central authority 152then calculates the policies that apply to that user 102 and sends thepolicy to the enforcement node 150 as a highly compressed bitmap.

Once downloaded, a tenant's policy is cached until a policy change ismade in the management system 120. When this happens, all of the cachedpolicies are purged, and the enforcement nodes 150 request the newpolicy when the user 102 next makes a request. In an embodiment, theenforcement node 150 exchange “heartbeats” periodically, so allenforcement nodes 150 are informed when there is a policy change. Anyenforcement node 150 can then pull the change in policy when it sees anew request.

The cloud-based system 100 can be a private cloud, a public cloud, acombination of a private cloud and a public cloud (hybrid cloud), or thelike. Cloud computing systems and methods abstract away physicalservers, storage, networking, etc., and instead offer these as on-demandand elastic resources. The National Institute of Standards andTechnology (NIST) provides a concise and specific definition whichstates cloud computing is a model for enabling convenient, on-demandnetwork access to a shared pool of configurable computing resources(e.g., networks, servers, storage, applications, and services) that canbe rapidly provisioned and released with minimal management effort orservice provider interaction. Cloud computing differs from the classicclient-server model by providing applications from a server that areexecuted and managed by a client's web browser or the like, with noinstalled client version of an application required. Centralizationgives cloud service providers complete control over the versions of thebrowser-based and other applications provided to clients, which removesthe need for version upgrades or license management on individual clientcomputing devices. The phrase “Software as a Service” (SaaS) issometimes used to describe application programs offered through cloudcomputing. A common shorthand for a provided cloud computing service (oreven an aggregation of all existing cloud services) is “the cloud.” Thecloud-based system 100 is illustrated herein as an example embodiment ofa cloud-based system, and other implementations are also contemplated.

As described herein, the terms cloud services and cloud applications maybe used interchangeably. The cloud service 106 is any service madeavailable to users on-demand via the Internet, as opposed to beingprovided from a company's on-premises servers. A cloud application, orcloud app, is a software program where cloud-based and local componentswork together. The cloud-based system 100 can be utilized to provideexample cloud services, including Zscaler Internet Access (ZIA), ZscalerPrivate Access (ZPA), and Zscaler Digital Experience (ZDX), all fromZscaler, Inc. (the assignee and applicant of the present application).The ZIA service can provide the access control, threat prevention, anddata protection described above with reference to the cloud-based system100. ZPA can include access control, microservice segmentation, etc. TheZDX service can provide monitoring of user experience, e.g., Quality ofExperience (QoE), Quality of Service (QoS), etc., in a manner that cangain insights based on continuous, inline monitoring. For example, theZIA service can provide a user with Internet Access, and the ZPA servicecan provide a user with access to enterprise resources instead oftraditional Virtual Private Networks (VPNs), namely ZPA provides ZeroTrust Network Access (ZTNA). Those of ordinary skill in the art willrecognize various other types of cloud services 106 are alsocontemplated. Also, other types of cloud architectures are alsocontemplated, with the cloud-based system 100 presented for illustrationpurposes.

Example Server Architecture

FIG. 2A is a block diagram of a server 200, which may be used in thecloud-based system 100, in other systems, or standalone. For example,the enforcement nodes 150 and the central authority 152 may be formed asone or more of the servers 200. The server 200 may be a digital computerthat, in terms of hardware architecture, generally includes a processor202, input/output (I/O) interfaces 204, a network interface 206, a datastore 208, and memory 210. It should be appreciated by those of ordinaryskill in the art that FIG. 2A depicts the server 200 in anoversimplified manner, and a practical embodiment may include additionalcomponents and suitably configured processing logic to support known orconventional operating features that are not described in detail herein.The components (202, 204, 206, 208, and 210) are communicatively coupledvia a local interface 212. The local interface 212 may be, for example,but not limited to, one or more buses or other wired or wirelessconnections, as is known in the art. The local interface 212 may haveadditional elements, which are omitted for simplicity, such ascontrollers, buffers (caches), drivers, repeaters, and receivers, amongmany others, to enable communications. Further, the local interface 212may include address, control, and/or data connections to enableappropriate communications among the aforementioned components.

The processor 202 is a hardware device for executing softwareinstructions. The processor 202 may be any custom made or commerciallyavailable processor, a Central Processing Unit (CPU), an auxiliaryprocessor among several processors associated with the server 200, asemiconductor-based microprocessor (in the form of a microchip orchipset), or generally any device for executing software instructions.When the server 200 is in operation, the processor 202 is configured toexecute software stored within the memory 210, to communicate data toand from the memory 210, and to generally control operations of theserver 200 pursuant to the software instructions. The I/O interfaces 204may be used to receive user input from and/or for providing systemoutput to one or more devices or components.

The network interface 206 may be used to enable the server 200 tocommunicate on a network, such as the Internet 104. The networkinterface 206 may include, for example, an Ethernet card or adapter or aWireless Local Area Network (WLAN) card or adapter. The networkinterface 206 may include address, control, and/or data connections toenable appropriate communications on the network. A data store 208 maybe used to store data. The data store 208 may include any of volatilememory elements (e.g., random access memory (RAM, such as DRAM, SRAM,SDRAM, and the like)), nonvolatile memory elements (e.g., ROM, harddrive, tape, CDROM, and the like), and combinations thereof. Moreover,the data store 208 may incorporate electronic, magnetic, optical, and/orother types of storage media. In one example, the data store 208 may belocated internal to the server 200, such as, for example, an internalhard drive connected to the local interface 212 in the server 200.Additionally, in another embodiment, the data store 208 may be locatedexternal to the server 200 such as, for example, an external hard driveconnected to the I/O interfaces 204 (e.g., SCSI or USB connection). In afurther embodiment, the data store 208 may be connected to the server200 through a network, such as, for example, a network-attached fileserver.

The memory 210 may include any of volatile memory elements (e.g., randomaccess memory (RAM, such as DRAM, SRAM, SDRAM, etc.)), nonvolatilememory elements (e.g., ROM, hard drive, tape, CDROM, etc.), andcombinations thereof. Moreover, the memory 210 may incorporateelectronic, magnetic, optical, and/or other types of storage media. Notethat the memory 210 may have a distributed architecture, where variouscomponents are situated remotely from one another but can be accessed bythe processor 202. The software in memory 210 may include one or moresoftware programs, each of which includes an ordered listing ofexecutable instructions for implementing logical functions. The softwarein the memory 210 includes a suitable Operating System (O/S) 214 and oneor more programs 216. The operating system 214 essentially controls theexecution of other computer programs, such as the one or more programs216, and provides scheduling, input-output control, file and datamanagement, memory management, and communication control and relatedservices. The one or more programs 216 may be configured to implementthe various processes, algorithms, methods, techniques, etc. describedherein.

Example User Device Architecture

FIG. 2B is a block diagram of a user device 250, which may be used withthe cloud-based system 100 or the like. Specifically, the user device250 can form a device used by one of the users 102, and this may includecommon devices such as laptops, smartphones, tablets, netbooks, personaldigital assistants, MP3 players, cell phones, e-book readers, IoTdevices, servers, desktops, printers, televisions, streaming mediadevices, and the like. The user device 250 can be a digital device that,in terms of hardware architecture, generally includes a processor 252,I/O interfaces 254, a network interface 256, a data store 258, andmemory 260. It should be appreciated by those of ordinary skill in theart that FIG. 2B depicts the user device 250 in an oversimplifiedmanner, and a practical embodiment may include additional components andsuitably configured processing logic to support known or conventionaloperating features that are not described in detail herein. Thecomponents (252, 254, 256, 258, and 252) are communicatively coupled viaa local interface 262. The local interface 262 can be, for example, butnot limited to, one or more buses or other wired or wirelessconnections, as is known in the art. The local interface 262 can haveadditional elements, which are omitted for simplicity, such ascontrollers, buffers (caches), drivers, repeaters, and receivers, amongmany others, to enable communications. Further, the local interface 262may include address, control, and/or data connections to enableappropriate communications among the aforementioned components.

The processor 252 is a hardware device for executing softwareinstructions. The processor 252 can be any custom made or commerciallyavailable processor, a CPU, an auxiliary processor among severalprocessors associated with the user device 250, a semiconductor-basedmicroprocessor (in the form of a microchip or chipset), or generally anydevice for executing software instructions. When the user device 250 isin operation, the processor 252 is configured to execute software storedwithin the memory 260, to communicate data to and from the memory 260,and to generally control operations of the user device 250 pursuant tothe software instructions. In an embodiment, the processor 252 mayinclude a mobile optimized processor such as optimized for powerconsumption and mobile applications. The I/O interfaces 254 can be usedto receive user input from and/or for providing system output. Userinput can be provided via, for example, a keypad, a touch screen, ascroll ball, a scroll bar, buttons, a barcode scanner, and the like.System output can be provided via a display device such as a LiquidCrystal Display (L.C. D), touch screen, and the like.

The network interface 256 enables wireless communication to an externalaccess device or network. Any number of suitable wireless datacommunication protocols, techniques, or methodologies can be supportedby the network interface 256, including any protocols for wirelesscommunication. The data store 258 may be used to store data. The datastore 258 may include any of volatile memory elements (e.g., randomaccess memory (RAM, such as DRAM, SRAM, SDRAM, and the like)),nonvolatile memory elements (e.g., ROM, hard drive, tape, CDROM, and thelike), and combinations thereof. Moreover, the data store 258 mayincorporate electronic, magnetic, optical, and/or other types of storagemedia.

The memory 260 may include any of volatile memory elements (e.g., randomaccess memory (RAM, such as DRAM, SRAM, SDRAM, etc.)), nonvolatilememory elements (e.g., ROM, hard drive, etc.), and combinations thereof.Moreover, the memory 260 may incorporate electronic, magnetic, optical,and/or other types of storage media. Note that the memory 260 may have adistributed architecture, where various components are situated remotelyfrom one another, but can be accessed by the processor 252. The softwarein memory 260 can include one or more software programs, each of whichincludes an ordered listing of executable instructions for implementinglogical functions. In the example of FIG. 2B, the software in the memory260 includes a suitable operating system 264 and programs 266. Theoperating system 264 essentially controls the execution of othercomputer programs and provides scheduling, input-output control, fileand data management, memory management, and communication control andrelated services. The programs 266 may include various applications,add-ons, etc. configured to provide end user functionality with the userdevice 250. For example, example programs 266 may include, but notlimited to, a web browser, social networking applications, streamingmedia applications, games, mapping and location applications, electronicmail applications, financial applications, and the like. In a typicalexample, the end-user typically uses one or more of the programs 266along with a network such as the cloud-based system 100.

Machine Learning in Network Security

Machine learning can be used in various applications, including malwaredetection, intrusion detection, threat classification, the user orcontent risk, detecting malicious clients or bots, etc. In a particularuse case, machine learning can be used on a content item, e.g., a file,to determine if further processing is required during inline processingin the cloud-based system 100. For example, machine learning can be usedin conjunction with a sandbox to identify malicious files. A sandbox, asthe name implies, is a safe environment where a file can be executed,opened, etc. for test purposes to determine whether the file ismalicious or benign. It can take a sandbox around 10 minutes before itis fully determined whether the file is malicious or benign.

Machine learning can determine a verdict in advance before a file issent to the sandbox. If a file is predicted as benign, it does not needto be sent to the sandbox. Otherwise, it is sent to the sandbox forfurther analysis/processing. Advantageously, utilizing machine learningto pre-filter a file significantly improves user experience by reducingthe overall quarantine time as well as reducing workload in the sandbox.Of course, machine learning cannot replace the sandbox since maliciousinformation from a static file is limited, while the sandbox can get amore accurate picture with dynamic behavior analysis. Further, itfollows that the machine learning predictions require high precision dueto the impact of a false prediction, i.e., finding a malicious file tobe benign.

In the context of inline processing, sandboxing does a great job indetecting malicious files, but there is a cost in latency, which affectsuser experience. Machine learning can alleviate this issue by giving anearlier verdict on the static files. However, it requires ML to haveextremely high precision, since the cost of a false positive and falsenegative are very high. For example, a benign hospital life-threateningfile, if mistakenly blocked due to an ML model's wrong verdict, wouldcause a life disaster. Similarly, undetected ransomware could causeproblems for an enterprise. Therefore, there is a need for ahigh-precision approach for both benign and malicious files.

The conventional approach to improve precision includes improving theprobability threshold to increase precision. A p-value (probabilityvalue) is a statistical assessment for measuring the reliability of aprediction, but this does not identify the unreliability of predictionswith high probabilities.

A description utilizing machine learning in the context of malwaredetection is described in commonly-assigned U.S. patent application Ser.No. 15/946,546, filed Apr. 5, 2018, and entitled “System and method formalware detection on a per packet basis,” the content of which isincorporated by reference herein. As described here, the typical machinelearning training process collects millions of malware samples, extractsa set of features from these samples, and feeds the features into amachine learning model to determine patterns in the data. The output ofthis training process is a machine learning model that can predictwhether a file that has not been seen before is malicious or not.

Decision Tree

In an embodiment, a generated machine learning model is a decision tree.A trained model may include a plurality of decision trees. Each of theplurality of decision trees may include one or more nodes, one or morebranches, and one or more termini. Each node in the trained decisiontree represents a feature and a decision boundary for that feature. Eachof the one or more termini is, in turn, associated with an outputprobability. Generally, each of the one or more nodes leads to anothernode via a branch until a terminus is reached, and an output score isassigned.

FIG. 3 is a diagram of a trained machine learning model 300. The machinelearning model 300 includes one or more features 310 and multiple trees320 a, 320 n. A feature is an individual measurable property orcharacteristic of a phenomenon being observed. The trees 320 a, 320 ncan be decision trees associated with a random forest or a gradientboosting decision trees machine learning model. In various embodiments,the trees 320 a, 320 b are constructed during training. While themachine learning model 300 is only depicted as having trees 320 a, 320n, in other embodiments, the machine learning model 300 includes aplurality of additional trees. The features 310, in the context ofmalicious file detection, relate to various properties orcharacteristics of the file.

The trees 320 a, 320 n include nodes 330 a, 330 b and termini 340 a, 340b, 340 c, 340 d. That is, the node 330 a is connected to termini 340 a,340 b and the node 330 b is connected to termini 340 c, 340, via one ormore branches. In other embodiments, the trees 320 a, 320 n include oneor more additional nodes, one or more additional branches, and one ormore additional termini. The nodes 330 each represent a feature and adecision boundary for that feature. The termini 340 can each beassociated with a probability of maliciousness, in the example ofmalicious file detection. Generally, each of the one or more nodes leadsto another node via a branch until a terminus is reached, and aprobability of maliciousness is assigned. The output of the trainedmachine learning model 300 is a weighted average of a probability ofmaliciousness predicted by each of the trees 320 a and the tree 320 n.

Ensemble Models

Multiple different machine learning models can be used as an ensemblemodel that obtain better predictive performance than could be obtainedfrom any of the constituent machine learning models alone. Theindividual models in an ensemble model could be tree-based (e.g., thedecision tree used by gradient boosting decision trees and randomforest) or neural networks or any other machine learning model where theprediction follows a decision path or activation path. For illustrationpurposes, the foregoing examples relate to decision trees.

The machine learning model 300 is an example of a decision tree. Adecision tree is a tool that uses a tree-like model of decisions andtheir possible consequences, including chance event outcomes, resourcecosts, and utility. It is one way to display an algorithm that onlycontains conditional control statements, i.e., if . . . then . . . else.Random forests or random decision forests are an ensemble model forclassification, regression and other tasks that operates by constructinga multitude of decision trees at training time and outputting the classthat is the mode of the classes (classification) or mean prediction(regression) of the individual trees. Random decision forests correctfor decision trees' habit of overfitting to their training set. Of note,each of the decision trees is independent of one another in the case ofRandom Forest. Gradient Boosting Decision Trees are dependent betweenone another. Gradient boosting is a machine learning technique forregression and classification problems, which produces a predictionmodel in the form of an ensemble of weak prediction models, typically adecision tree.

Blind Spots

Instance-based machine learning approaches make predictions based on itsneighbor, that is, the examples similar to it. On the other hand, if nosimilar examples are surrounding the example under prediction, there isinsufficient support for the prediction. Thus, the prediction isuntrustworthy. An instance-based approach needs a similarity thresholdto decide whether there are similar examples. However, the similarity isrelative and not absolute. The similarity is also feature dependent.Again, blind spots in a machine learning model are regions in a featurespace defined by ensemble trees where there is insufficient orconflicting evidence from previously seen data (e.g., training). Blindspots are the target of adversarial attacks where the models are fooledwith malicious input. Machine learning models are unable to makeaccurate predictions at blind spots.

For an example of a blind spot, is broccoli more similar to caulifloweror kale? It is clear that from the shape perspective, broccoli is closerto cauliflower. While if the green color is the dominant feature, thenbroccoli becomes closer to kale. Thus, this model requires additionalfeatures as shape and color alone are not sufficient for distinguishingexamples.

Prudent Ensemble Models

The present disclosure includes measuring the reliability of aprediction to provide confidence/over prediction. These reliabilitymeasures can also be double-checked and tracked to improve themeasurement of reliability further. For example, in the maliciouscontent item detection use case, the unreliability predictions could bedoubled checked by a sandbox. The unreliability predictions can increaseprecision by filtering out unreliable predictions. If a prediction ismade, it has very high precision. While for those, it is not sure, theycan be analyzed further to identify malware concept drift or discrepancyin the data distribution.

Prudent Ensemble Model Process

FIG. 4 is a flowchart of a prudent ensemble model process 520. Theprocess 520 includes training an ensemble model (step 522). This stepproceeds as is known in the machine learning art. As described herein,the ensemble model could be tree-based (e.g., the decision tree used bygradient boosting decision trees and random forest) or neural networksor any other machine learning model where the prediction follows adecision path or activation path.

The process 520 includes determining blind spots in the trained ensemblemodel (step 524). Again, a blind spot is a location where the trainedensemble model has not seen any examples with the combination of thefeatures at the location or has examples with conflicting labels. Thedetermined blind spots are marked or otherwise noted (step 526). Thetrained ensemble model is utilized in production to make predictions,but any predictions that are in marked blind spots are filtered out(ignored) as being unreliable (step 528).

Again, by filtering out unreliable predictions, that is, the predictionsthat fall into blind spots, the process 520 counters adversarialattacks, including those not just on decision boundary attack, but alsothose far away from the decision boundary. For example, a malicious filecan be configured to fool the model by having characteristics similar toa benign content item, but still being malicious. The process 520advantageously protects against such attacks as the malicious file thattries to fool the model will end up in a blind spot as such as filewould not have existing examples. Accordingly, this file would berejected due to the blind spot. The process 520 further achievesskyscraper high precision, and the process 520 increases the visibilityof the trained ensemble model by explicitly exposing the vulnerable partof the model. The vulnerable part of the model can be improved throughfurther training. The process 520 leverages the idea from instance-based(e.g., k-nearest neighbor) and integrates it into ensemble models toenhance their predictions.

The trained ensemble model uses learned models to define what aresimilar examples. Ensemble models non-linearly segment the feature spaceinto small regions. Each region is the result of superimposing thedecision paths from all sub-models. Examples within the same region aredeemed similar. If the prediction paths for an example fall into aregion where no examples have been seen previously or only examples withconflicting labels, that means it is a region without sufficient supportfrom examples, thus named blind spots. The blind spots defined in thisway can be anywhere in the feature space and do not have to near thedecision boundary.

By filtering out predictions fell into blind spots, the process 520 cancounter the adversary attack in various regions in feature space (notjust those close to decision boundary). This is complementary toexisting solutions for the adversarial attack. Since the adversarialexamples generated using the existing data are limited. There are stillblind spots remaining after the hardening of the model trained withadversary examples, especially those not close to the decision boundary.

Content Processing Process by an Inline Security System

FIG. 5 is a flowchart of a content processing process 540, implementedby the cloud-based security system 100. The process 540 can includeobtaining a trained machine learning ensemble model to identifymalicious content items (step 542). The trained machine learningensemble model can be from the process 520. The process 540 includesreceiving a content item between a user device and a location on theInternet or an enterprise network (step 544), utilizing the trainedmachine learning ensemble model to determine whether the content item ismalicious (step 546), responsive to the trained machine learningensemble model determining the content item is malicious or determiningthe content item is benign but such determining is in a blind spot ofthe trained ensemble model, performing further processing on the contentitem (step 548), and, responsive to the trained machine learningensemble model determining the content item is benign with suchdetermination not in a blind spot of the trained machine learningensemble model, allowing the content item (step 550).

As mentioned, the blind spot is a location where the trained machinelearning ensemble model has not seen any examples with a combination offeatures at the location or has examples with conflicting labels. Theprocess 540 can further include training the trained machine learningensemble model to identify malicious content items and identifying andmarking blind spots in the trained machine learning ensemble model. Theprocess 540 can further include, subsequent to the further processing,one of allowing the content item and blocking the content item based onthe further processing.

Further processing can include performing a dynamic analysis of thecontent item in a sandbox. For example, this can include the analysisdescribed in U.S. Pat. No. 9,152,789, issued on Oct. 6, 2015, andentitled “Systems and methods for dynamic cloud-based malware behavioranalysis,” the contents of which are incorporated by reference herein.In an embodiment, the further processing includes blocking the contentitem in the cloud-based security system based on a classification by thetrained machine learning ensemble model. Here, the trained machinelearning ensemble model can be viewed as the final decision withoutrequiring a sandbox or the like.

In an embodiment, the content item is malicious and configured to foolthe trained machine learning ensemble model via an adversarial attackwhere the content item is configured to mimic benign features, andwherein the content item lands on a blind spot in the trained machinelearning ensemble model thereby preventing the adversarial attack. Thecontent item can be one of an executable file, a Portable Document File(PDF) file, a Microsoft Office file, and a JavaScript file. Thecloud-based security system can be located inline between the userdevice and the location.

Smart Quarantine Approach

The present disclosure includes a smart quarantine approach wheremachine learning is utilized as a front-end to a scanning system todecide whether or not to scan a particular file. Again, the goal in suchan approach is to minimize waiting time, risk, and cost. Of note, thesmart quarantine approach is described herein with reference to thecloud-based system 100, offering a cloud security service. Those skilledin the art will recognize the smart quarantine approach contemplates usein other architectures, including in a stand-along software programexecuted on the user device 300, in a security appliance, in a router,in a Secure Web Gateway (SWG), in a Web proxy, etc.

Conventional Quarantine Process

FIG. 6 is a flow diagram of a conventional quarantine process 600 forquarantining, scanning, blocking, and allowing a file 602. The file 602can be a document (e.g., a Microsoft Office document or the like), aPortable Document Format (PDF), or an executable file (e.g., a PortableExecutable (P.E.) file in 32 or 64-bit format). The file 602 isobtained, and then policy 604 determinates how the file 602 isprocessed, namely either quarantined (step 604-1), allowed and scanned(step 604-2), or allowed and not scanned (step 604-3). In thecloud-based system 100, the actions of quarantine and scanning may beseparate. For example, a file may be blocked to the end user 102 in thecloud-based system 100 if it is held, i.e., quarantined (step 606). Thefile may be allowed to the end user 102 and simultaneously scanned by asandbox (steps 604-2, 608). The result of the sandbox 608 is a score,and it can be used to determine whether the file 602 is malicious orbenign (step 610). Again, the sandbox 608 is configured to run the file602 in a controlled environment (i.e., a “sandbox”) and performobservation and analysis to determine behavior. For example, there canbe a scoring threshold, X, and a score above it means the file 602 isdetermined to be malicious (step 612), and a score below means the file602 is determined to be benign (step 614).

The step 604-3 immediately allows the file 602 to the end user 102. Thestep 604-1 holds the file 602 (step 606), and the step 604-2 immediatelyallows the file 602 to the end user 102, but still performs scanning inthe sandbox 608. For example, if the file 602 is malicious (step 612),but allowed at the step 604-2, the file 602 can be blocked the nexttime. If the file 602 is held (step 606) and the file 602 is malicious(step 612), the file 602 can be blocked, such as in the cloud-basedsystem 100. If the file 602 is held (step 606) and found to be benign(step 614), the file 602 can be allowed to the end user 102.

Machine Learning Smart Quarantining Process

FIG. 7 is a flow diagram of a smart quarantine process 700A forquarantining, scanning, blocking, and allowing a file 602, where machinelearning 702 is used to front end whether or not to hold the file 602.FIG. 8 is a flow diagram of a smart quarantine process 700B forquarantining, scanning, blocking, and allowing a file 602, where machinelearning 702 is used to front end whether or not to hold or scan thefile 602. FIG. 9 is a flow diagram of a smart quarantine process 700Cfor quarantining, scanning, blocking, and allowing a file 602, wheremachine learning 702 is used to front end whether or not to hold, scan,or allow the file 602.

In FIG. 7 , the smart quarantine process 700A includes obtaining thefile 602 and then policy 704 determinates how the file 602 is processed,either quarantined (step 704-1), processed by the machine learning 702to determine whether to quarantine (step 704-2) or to allow and scan(step 704-3), or allowed without a scan (step 704-4). That is, in thesmart quarantine process 700A, the machine learning 702 is used to frontend the allow and scan step. Thus, in the smart quarantine process 700A,the allow and scan is now augmented to become quarantine if maliciousfrom the machine learning 702 (step 704-2) or allow and scan if notmalicious from the machine learning 702 (step 704-3).

Similar to the quarantine process 600, the smart quarantine process 700Aincludes holding the file 602 after the steps 704-1, 704-2 (step 706),and scanning the file 602 with a sandbox 708 after the steps 704-1,704-2, 704-3. The sandbox 708 scores the file 602 (step 710), and thesmart quarantine process 700A determines if the file 602 is malicious(step 712) or benign (step 714) based thereon. If the file was held(step 706), the smart quarantine process 700A can block the file 602 ifmalicious. Further, the smart quarantine process 700A also includes thestep 704-4 of allowing without a scan based on the policy 704.

In FIG. 8 , the smart quarantine process 700B includes combining themachine learning 702 with the policy 704. Here, the machine learning 702front ends both the decision to quarantine and to allow and scan, notjust the decision to allow and scan. Here, the smart quarantine process700B can include three outputs of the combined machine learning 702 andpolicy 704, namely quarantine if the machine learning 702 determines thefile 602 is malicious (step 720-1), allow and scan if the machinelearning 702 determines the file 602 is not malicious (step 720-2), andallow without a scan if the policy 704 dictates for the file 602 (step720-3). The remainder of the steps in the smart quarantine process 700Bare the same as in the smart quarantine process 700A.

In FIG. 9 , the smart quarantine process 700C also includes combiningthe machine learning 702 with the policy 704, but here the machinelearning 702 output is used in all three decisions. The machine learning702 front ends all the decisions, namely, quarantine if the machinelearning 702 determines the file 602 is malicious (step 730-1), allowand scan if the machine learning 702 determines the file 602 is notmalicious (step 730-2), and allow without a scan if the policy 704dictates for the file 602 and if the machine learning 702 determines thefile is benign (step 730-3). The remainder of the steps in the smartquarantine process 700C are the same as in the smart quarantine process700A, 700B.

The machine learning 702 can include any of the techniques describedherein. The policy 704 can be determined by a tenant associated with theuser 102. For example, the policy 704 can be based on a type of the file602, e.g., quarantine all executables, allow and scan all documents andPDFs, etc. The policy 704 can also be based on other factors such asuser location, the user device 250 types, network access technique, etc.

The smart quarantine process 700A, 700B, 700C utilizes machine learningas a front end to decide whether or not to hold the file 602 (FIG. 7 ),whether or not to hold or scan the file 602 (FIG. 8 ), and whether ornot to hold, scan, or allow the file 602 (FIG. 9 ). The smart quarantineprocesses 700A, 700B, 700C address the three metrics describedherein—waiting time, cost, and risk. Specifically, the conventionalquarantine process 600 has a baseline for each of these metrics. Thesmart quarantine processes 700A, 700B, 700C improves all of thesemetrics relative to the conventional quarantine process 600.

The smart quarantine processes 700A reduces risk relative to theconventional quarantine process 600 by utilizing the machine learning702 to augment and improve the allow and scan step. Allow and scan isrequired for some files as the users 102 simply do not want every file602 held for the sandbox 708. Thus, allow and scan poses some risk. Themachine learning 702 can reduce this risk such that some of the files602 that would be allowed and scanned are now held based on thedetermination of the machine learning 702.

The smart quarantine processes 700B both reduces the risk and thewaiting time relative to the conventional quarantine process 600 byutilizing the machine learning 702 to augment and improve the allow andscan step and the quarantine step. Here, the smart quarantine processes700B provides the same benefits as the smart quarantine processes 700Afor the allow and scan step. Additionally, the smart quarantineprocesses 700B only holds the files 602 if the output of the machinelearning 702 determines the file 602 is malicious, thereby reducing thenumber of files 602 that are held.

Finally, the smart quarantine processes 700C reduces the waiting time,the cost, and the risk relative to the conventional quarantine process600 by utilizing the machine learning 702 to augment and improve theallow and scan step, the quarantine step, and the allow without scanstep. Again, the smart quarantine processes 700C has the same benefitsas the smart quarantine processes 700A, 700B. Additionally, the smartquarantine processes 700C further augments the allow without scan onlywhere the machine learning 702 determines the file 602 is benign. Also,the sandbox is minimized as it only handles files determined assuspicious in the machine learning 702. Thus, the smart quarantineprocesses 700C reduces processing costs—the sandbox only has to addresssuspicious files 602.

Experimental Results

The following table illustrates a set of data from actual monitoring inthe cloud-based system 100, using the conventional quarantine process600. Here, the files 602 include documents (M.S. document), PDF files,and PE 32/64 files. In the quarantine step 604-1, a total of 217,961files were held, but only 1900 were malicious. Also, for the allow andscan step 604-2, a total of 2538 files were allowed, but eventuallydetermined to be malicious after the scanning.

policy File type malicious benign subtotal quarantined MS document 959208 59217 PDF 0 104905 104905 PE 32/64 1891 51948 53839 All 1900216061 217961 Allow and scan MS document 83 PDF 0 PE 32/64 2455 All 2538

The following table illustrates the same set of data with theintroduction of the machine learning 702. As can be seen in the abovetable, a lot of files 602 are held and scanned that may not have beennecessary. In the below table, the machine learning 702 provides TruePositives (T.P.) and False Positives (F.P.). Here, the machine learning702 determines that only 3806 files should be held in quarantine, not217,961.

policy File type ML TP ML FP subtotal All MS document 84 216 300 PDF 022 22 PE 32/64 2973 511 3484 All 3057 749 3806 Allow and scan MSdocument 60 182 242 PDF 0 17 17 PE 32/64 1366 253 1619 All 1426 452 1878Model Evolution

To describe the evolution of machine learning models, some examplemachine learning models include Random Forest, XGBoost, and LightGBM. Ofcourse, other types of models are also contemplated herein. The RandomForest includes a software implementation around 2017 with a large modelsize, e.g., on the order of 600 Mb. XGBoost includes open sourcesoftware providing a gradient boosting framework and a softwareimplementation around 2018 includes a model size on the order of 100 Mb,but is limited to on the order of 15 million training samples. Finally,LightGBM also provides gradient boosting framework having a similardetection and False Positive (FP) rate as XGBoost, but with about athird the training time and a software implementation from 2019 includesabout half the size of the XGBoost model size.

Gradient boosting is a machine learning technique for regression andclassification problems, which produces a prediction model in the formof an ensemble of weak prediction models, typically decision trees. Itbuilds the model in a stage-wise fashion like other boosting methods do,and it generalizes them by allowing optimization of an arbitrarydifferentiable loss function.

Model Serving

The present disclosure provides machine learning model abstraction forruntime. There are a couple drivers for the machine learning modelabstraction. First, machine learning models can use differentalgorithms, such as in the example above, RandomForest→XGBoost→LightGBM. Each algorithm can include its own runtimelibrary. The machine learning model abstraction layer can provide asingle runtime library for multiple different algorithms. Second, eachof these algorithms, in their software implementations, include traininginformation that is not relevant to runtime. Examples of traininginformation can include hyperparameters which are used to control thetraining process. Third, the machine learning model abstraction layercan abstract the algorithm into a tree structure that is very fast formodel serving.

The machine learning model abstraction layer is algorithm andprogramming language agnostic and can be in a portable format for anycomputer architecture. The machine learning model abstraction layerdecouples the training process from the runtime process.

In an embodiment, the present disclosure utilizes the LightGBM model.LightGBM is a gradient boosting framework that uses tree-based learningalgorithms. The LightGBM model includes metadata that is extra overhead.FIG. 10 is a screenshot of an example of the metadata that is extraoverhead. For example, an example of the extra overhead includes featurenames, feature info, tree sizes, etc. The LightGBM model also includesinformation that is not useful at runtime. FIG. 11 is a screenshot of anexample of information that is not useful at runtime. For example,split_gain, leaf_count, internal_value, and internal_count are notuseful at runtime and the decision_type is duplicated. Finally, theLightGBM model includes parameters that are useful at understanding howthe model was trained, but serve no purpose at runtime (model serving).FIG. 12 is a screenshot of parameters that are useful at understandinghow the model was trained, but serve no purpose at runtime.

In a further embodiment, the machine learning model abstraction layercan remove features from the machine learning model, namely featuresthat are not used. This removes any “holes” in the model that wastespace and eliminates the need to regenerate the feature set.

In another embodiment, the machine learning model abstraction layer caninclude normalizing model probabilities into a machine learning score,such that scores are consider across model changes. Of note, each modeltrained is unique and has its own unique thresholds. The machinelearning model abstraction layer can pick the best thresholds for agiven model and normalize probabilities. For example, a benign score canbe below 40, a suspicious score can be between 40 and 70, and amalicious score can be above 70. Of course, other embodiments are alsocontemplated.

Model Abstraction Process

FIG. 13 is a flowchart of a machine learning abstraction process 800.The machine learning abstraction process 800 can be acomputer-implemented method, embodied as instructions stored innon-transitory computer readable medium, and implemented via anapparatus such as the server 200.

The machine learning abstraction process 800 includes training a machinelearning model with data for identifying features in monitored trafficin a network (step 802); analyzing the trained machine learning model toidentify information overhead therein, wherein the information overheadis utilized in part for the training (step 804); removing theinformation overhead in the machine learning model (step 806); andproviding the machine learning model for runtime use for identifying thefeatures in the monitored traffic, with the removed information overheadfrom the machine learning model (step 808).

The machine learning abstraction process 800 can further includeidentifying features that are not used in the trained machine learningmodel; and removing the identified features prior to the providing. Themachine learning abstraction process 800 determining thresholds for theidentifying features in the trained machine learning model; andnormalizing the thresholds to a scoring system.

The information overhead can include hyperparameters. The informationoverhead can include metadata that is extra overhead in the trainedmachine learning model. The information overhead can include informationfrom the training that is not useful at runtime in the trained machinelearning model. The information overhead can include parameters that areused to understand the training.

The machine learning model can include a gradient boosting frameworkthat uses tree-based learning algorithms. The providing step 808 can beto a cloud-based system that utilizes the machine learning model forinline monitoring of the monitored traffic.

It will be appreciated that some embodiments described herein mayinclude one or more generic or specialized processors (“one or moreprocessors”) such as microprocessors; Central Processing Units (CPUs);Digital Signal Processors (DSPs): customized processors such as NetworkProcessors (NPs) or Network Processing Units (NPUs), Graphics ProcessingUnits (GPUs), or the like; Field Programmable Gate Arrays (FPGAs); andthe like along with unique stored program instructions (including bothsoftware and firmware) for control thereof to implement, in conjunctionwith certain non-processor circuits, some, most, or all of the functionsof the methods and/or systems described herein. Alternatively, some orall functions may be implemented by a state machine that has no storedprogram instructions, or in one or more Application-Specific IntegratedCircuits (ASICs), in which each function or some combinations of certainof the functions are implemented as custom logic or circuitry. Ofcourse, a combination of the aforementioned approaches may be used. Forsome of the embodiments described herein, a corresponding device inhardware and optionally with software, firmware, and a combinationthereof can be referred to as “circuitry configured or adapted to,”“logic configured or adapted to,” etc. perform a set of operations,steps, methods, processes, algorithms, functions, techniques, etc. ondigital and/or analog signals as described herein for the variousembodiments.

Moreover, some embodiments may include a non-transitorycomputer-readable storage medium having computer-readable code storedthereon for programming a computer, server, appliance, device,processor, circuit, etc. each of which may include a processor toperform functions as described and claimed herein. Examples of suchcomputer-readable storage mediums include, but are not limited to, ahard disk, an optical storage device, a magnetic storage device, aRead-Only Memory (ROM), a Programmable Read-Only Memory (PROM), anErasable Programmable Read-Only Memory (EPROM), an Electrically ErasableProgrammable Read-Only Memory (EEPROM), Flash memory, and the like. Whenstored in the non-transitory computer-readable medium, software caninclude instructions executable by a processor or device (e.g., any typeof programmable circuitry or logic) that, in response to such execution,cause a processor or the device to perform a set of operations, steps,methods, processes, algorithms, functions, techniques, etc. as describedherein for the various embodiments.

Although the present disclosure has been illustrated and describedherein with reference to preferred embodiments and specific examplesthereof, it will be readily apparent to those of ordinary skill in theart that other embodiments and examples may perform similar functionsand/or achieve like results. All such equivalent embodiments andexamples are within the spirit and scope of the present disclosure, arecontemplated thereby, and are intended to be covered by the followingclaims.

What is claimed is:
 1. A non-transitory computer-readable storage mediumhaving computer-readable code stored thereon for programming a node in acloud-based system to perform steps of: receiving a trained machinelearning model that has been processed with training information removedtherefrom, wherein the training information is utilized in training ofthe trained machine learning model, and wherein the training informationis not relevant to runtime, including features not used in the trainedmachine learning model; monitoring traffic, inline at the node,including processing the traffic with the trained machine learningmodel; obtaining a verdict on the traffic based on the trained machinelearning model; and performing an action on the traffic based on theverdict.
 2. The non-transitory computer-readable storage medium of claim1, wherein the verdict is associated with malware detection and theaction includes whether or not to sandbox content in the traffic.
 3. Thenon-transitory computer-readable storage medium of claim 1, wherein theverdict is associated with intrusion detection and the action includeswhether to allow or block some of the traffic.
 4. The non-transitorycomputer-readable storage medium of claim 1, wherein the verdict isassociated with threat classification and the action includes whether toallow or block content in the traffic.
 5. The non-transitorycomputer-readable storage medium of claim 1, wherein the traininginformation includes parameters that are useful at understanding how thetrained machine learning model was trained, but serve no purpose atruntime.
 6. The non-transitory computer-readable storage medium of claim1, wherein the training information includes hyperparameters used tocontrol training.
 7. The non-transitory computer-readable storage mediumof claim 1, wherein the training information includes metadata that isextra overhead in the trained machine learning model.
 8. Thenon-transitory computer-readable storage medium of claim 1, wherein thetraining information includes any of feature names, feature info, andtree sizes in the trained machine learning model.
 9. The non-transitorycomputer-readable storage medium of claim 1, wherein the trained machinelearning model includes a gradient boosting framework that usestree-based learning algorithms.
 10. A method comprising: receiving atrained machine learning model that has been processed with traininginformation removed therefrom, wherein the training information isutilized in training of the trained machine learning model, and whereinthe training information is not relevant to runtime, including featuresnot used in the trained machine learning model; monitoring traffic,inline at the node, including processing the traffic with the trainedmachine learning model; obtaining a verdict on the traffic based on thetrained machine learning model; and performing an action on the trafficbased on the verdict.
 11. The method of claim 10, wherein the verdict isassociated with malware detection and the action includes whether or notto sandbox content in the traffic.
 12. The method of claim 10, whereinthe verdict is associated with intrusion detection and the actionincludes whether to allow or block some of the traffic.
 13. The methodof claim 10, wherein the verdict is associated with threatclassification and the action includes whether to allow or block contentin the traffic.
 14. The method of claim 10, wherein the traininginformation includes parameters that are useful at understanding how thetrained machine learning model was trained, but serve no purpose atruntime.
 15. The method of claim 10, wherein the training informationincludes hyperparameters used to control training.
 16. The method ofclaim 10, wherein the training information includes metadata that isextra overhead in the trained machine learning model.
 17. The method ofclaim 10, wherein the training information includes any of featurenames, feature info, and tree sizes in the trained machine learningmodel.
 18. The method of claim 10, wherein the trained machine learningmodel includes a gradient boosting framework that uses tree-basedlearning algorithms.